package com.chatTools.client;

import com.chatTools.message.*;
import com.chatTools.protocol.MessageCodecSharable;
import com.chatTools.protocol.ProtocolFrameDecoder;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;
import io.netty.handler.timeout.IdleStateHandler;
import lombok.extern.slf4j.Slf4j;

import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Scanner;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicBoolean;

/**
 * @author： pang-yun
 * @date： 2021-09-16 17:30
 * 功能目的 ：用于 客户端代码
 */

/**
 * 两个线程之间的通信   CountDownLatch
 * //  用于线程之间的通信
 * //CountDownLatch是一个同步工具类，用来协调多个线程之间的同步，或者说起到线程之间的通信（而不是用作互斥的作用）
 * CountDownLatch WAIT_FOR_LOGIN =new CountDownLatch(1);
 * WAIT_FOR_LOGIN.countDown();
 * WAIT_FOR_LOGIN.await();
 * <p>
 * AtomicBoolean  LOGIN =new AtomicBoolean();  多线程情况下 线程安全
 * // 使用原子类来 实现线程之间的通信
 */


@Slf4j
public class ChatClient {


    public static void main(String[] args) {
        NioEventLoopGroup worker = new NioEventLoopGroup();

        // 各类处理器
        LoggingHandler LOGGING_HANDLER = new LoggingHandler(LogLevel.INFO);
        MessageCodecSharable MESSAGE_CODEC_SHARABLE = new MessageCodecSharable();
        ProtocolFrameDecoder PROTOCOL_DECODER = new ProtocolFrameDecoder();


        //  用于线程之间的通信
        //CountDownLatch是一个同步工具类，用来协调多个线程之间的同步，或者说起到线程之间的通信（而不是用作互斥的作用）
        CountDownLatch WAIT_FOR_LOGIN = new CountDownLatch(1);

        AtomicBoolean LOGIN = new AtomicBoolean();
        try {
            ChannelFuture channelFuture = new Bootstrap()
                    .group(worker)
                    .channel(NioSocketChannel.class)
                    .handler(new ChannelInitializer<NioSocketChannel>() {
                        @Override
                        protected void initChannel(NioSocketChannel ch) throws Exception {

                            //------------------------------帧处理器 编解码 日志等 处理类----------------------------
                            // 首先 帧处理器  接收完整消息
                            ch.pipeline().addLast(PROTOCOL_DECODER);
                            // 日志
                            //  ch.pipeline().addLast(LOGGING_HANDLER);
                            // 编解码
                            ch.pipeline().addLast(MESSAGE_CODEC_SHARABLE);

                            //---------------------------为防止假死 所做的心跳处理类---------------------------

                            //假死时间处理 判断是否 读写时间过长   (read write all)
                            // 3s 之内如果没有 write 数据， 就会触发一个 IdleState.WRITE_IDLE 事件
                            ch.pipeline().addLast(new IdleStateHandler(0, 3, 0));
                            // 触发假死 事件之后来进行处理    ChannelDuplexHandler  双向处理器
                            ch.pipeline().addLast(new ChannelDuplexHandler() {
                                // trigger 触发器     用来触发特殊事件
                                @Override
                                public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
                                    IdleStateEvent event = (IdleStateEvent) evt;
                                    // 出发了空闲事件
                                    if (event.state() == IdleState.WRITER_IDLE) {
                                        //  log.info("客户端已经 3s 没有写数据了，发送一个心跳包！！！");
                                        ctx.writeAndFlush(new PingMessage());
                                    }
                                }
                            });
                            //----------------------------------------------------------------------------------------
                            ch.pipeline().addLast("client handler", new ChannelInboundHandlerAdapter() {
                                //接收响应消息
                                @Override
                                public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
                                    log.info("msg : {}", msg);
                                    // 处理登录的 逻辑
                                    if (msg instanceof LoginResponseMessage) {
                                        LoginResponseMessage responseMessage = (LoginResponseMessage) msg;
                                        if (responseMessage.isSuccess()) {
                                            // 如果登录成功
                                            LOGIN.set(true);
                                        }
                                        // 环形 system in 线程
                                        WAIT_FOR_LOGIN.countDown();
                                    }


                                }

                                //在连接建立之后 触发 active事件
                                @Override
                                public void channelActive(ChannelHandlerContext ctx) throws Exception {
                                    // 因为主线程会经常阻塞  我们需要新开一个线程来 接收用户在控制台的输入  负责向服务器发送各种消息
                                    new Thread(() -> {
                                        Scanner scanner = new Scanner(System.in);
                                        System.out.println("请输入 username :");
                                        String username = scanner.nextLine();
                                        System.out.println("请 输入 password :");
                                        String password = scanner.nextLine();
                                        // 构造消息对象
                                        LoginRequestMessage loginRequestMessage = new LoginRequestMessage(username, password);
                                        // 发送消息
                                        /*
                                         * 这是 入站处理器  写入内容后 会触发出站操作   出站 从后向前 寻找 handler
                                         *   经 MessageCodecSharable 后 进行 编码
                                         *   后 LoggingHandler  记录日志
                                         * */
                                        ctx.writeAndFlush(loginRequestMessage);

                                        System.out.println("waiting.....");

                                        // 使用 countdownlatch   来 确保 得到消息之后  countdownlatch 变为0 之后  该线程可以继续执行
                                        try {
                                                                                                                                                                                                                                                                                                                                                                                                                                                                            WAIT_FOR_LOGIN.await();
                                        } catch (InterruptedException e) {
                                            e.printStackTrace();
                                        }
                                      /*  try {
                                            System.in.read();
                                        } catch (IOException e) {
                                            e.printStackTrace();
                                        }*/

                                        // 如果登录失败
                                        if (!LOGIN.get()) {
                                            ctx.channel().close();
                                            System.out.println("----------登录失败---------------");
                                            return;
                                        }

                                        // 登录成功之后  开始其他选项
                                        while (true) {
                                            System.out.println("============ 功能菜单 ============");
                                            System.out.println("send [username] [content]");
                                            System.out.println("gsend [group name] [content]");
                                            System.out.println("gcreate [group name] [m1,m2,m3...]");
                                            System.out.println("gmembers [group name]");
                                            System.out.println("gjoin [group name]");
                                            System.out.println("gquit [group name]");
                                            System.out.println("quit");
                                            System.out.println("==================================");

                                            // 用 sacnner 将其阻塞
                                            String command = scanner.nextLine();
                                            // 对输入的 参数 以 空格进行分隔
                                            final String[] s = command.split(" ");

                                            // 以第一个参数 区分命令
                                            switch (s[0]) {
                                                case "send":
                                                    // 发送一条聊天消息   from  to  content
                                                    ctx.writeAndFlush(new ChatRequestMessage(username, s[1], s[2]));
                                                    break;
                                                case "gsend":
                                                    ctx.writeAndFlush(new GroupChatRequestMessage(username, s[1], s[2]));
                                                    break;
                                                case "gcreate":
                                                    String[] split = s[2].split(",");
                                                    List<String> list = Arrays.asList(split);
                                                    HashSet<String> set = new HashSet<>(list);
                                                    ctx.writeAndFlush(new GroupCreateRequestMessage(s[1], set));
                                                    break;
                                                case "gmembers":
                                                    ctx.writeAndFlush(new GroupMemberRequestMessage(s[1]));
                                                    break;
                                                case "gjoin":
                                                    ctx.writeAndFlush(new GroupJoinRequestMessage(username, s[1]));
                                                    break;
                                                case "gquit":
                                                    ctx.writeAndFlush(new GroupQuitRequestMessage(username, s[1]));
                                                    break;
                                                case "quit":
                                                    ctx.channel().close();
                                                    return;

                                            }

                                        }
                                    }, "system in").start();
                                }
                            });

                        }
                    }).connect("localhost", 9999).sync();
            channelFuture.channel().closeFuture().sync();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            worker.shutdownGracefully();
        }
    }
}
